package com.dify.knowledge.repository;

import com.baomidou.mybatisplus.core.mapper.BaseMapper;
import com.dify.knowledge.entity.DocumentVersion;
import org.apache.ibatis.annotations.Mapper;
import org.apache.ibatis.annotations.Param;
import org.apache.ibatis.annotations.Select;
import org.apache.ibatis.annotations.Update;

import java.util.List;

/**
 * 文档版本Mapper接口
 */
@Mapper
public interface DocumentVersionMapper extends BaseMapper<DocumentVersion> {
    
    /**
     * 根据文档ID查询所有版本
     * @param documentId 文档ID
     * @return 版本列表
     */
    @Select("SELECT * FROM document_version WHERE document_id = #{documentId} ORDER BY version_number DESC")
    List<DocumentVersion> selectByDocumentId(@Param("documentId") Long documentId);
    
    /**
     * 获取文档的最大版本号
     * @param documentId 文档ID
     * @return 最大版本号
     */
    @Select("SELECT MAX(version_number) FROM document_version WHERE document_id = #{documentId}")
    Integer selectMaxVersionByDocumentId(@Param("documentId") Long documentId);
    
    /**
     * 获取文档的当前启用版本
     * @param documentId 文档ID
     * @return 启用版本
     */
    @Select("SELECT * FROM document_version WHERE document_id = #{documentId} AND is_active = TRUE LIMIT 1")
    DocumentVersion selectActiveVersionByDocumentId(@Param("documentId") Long documentId);
    
    /**
     * 取消文档的所有版本启用状态
     * @param documentId 文档ID
     * @return 更新行数
     */
    @Update("UPDATE document_version SET is_active = FALSE WHERE document_id = #{documentId}")
    int deactivateAllVersionsByDocumentId(@Param("documentId") Long documentId);
    
    /**
     * 启用指定版本
     * @param versionId 版本ID
     * @return 更新行数
     */
    @Update("UPDATE document_version SET is_active = TRUE WHERE id = #{versionId}")
    int activateVersion(@Param("versionId") Long versionId);
}
